﻿﻿using System.Collections.Generic;
using System.Data;
using System.Linq;
using Viyi.Util.Data.Converter;

namespace Viyi.Util.Data
{
    /// <summary>
    /// 对<see cref="DataTable"/>进行扩展，以方便的处理数据表中的列
    /// </summary>
    public static class DataTableUtil
    {
        /// <summary>
        /// 根据枚举将单元格值转换为枚举名称。
        /// 如果newColumnName为null，则使用新的值替换旧的值（新列换旧列）
        /// 否则为新值创建新的列
        /// </summary>
        /// <typeparam name="T"></typeparam>
        /// <param name="dataTable"></param>
        /// <param name="columnName"></param>
        /// <param name="newColumnName"></param>
        public static void MapEnum<T>(this DataTable dataTable,
            string columnName, string newColumnName = null)
        {
            new EnumColumnConverter(typeof(T)).Convert(dataTable, columnName, newColumnName);
        }

        /// <summary>
        /// 根据指定的格式字符串，将列数据转换成字符串。
        /// 格式模板规则与string.Format()的格式模板规则相同。
        /// </summary>
        /// <param name="dataTable"></param>
        /// <param name="formatPattern"></param>
        /// <param name="columnName"></param>
        /// <param name="newColumnName"></param>
        public static void FormatColumn(this DataTable dataTable,
            string formatPattern,
            string columnName, string newColumnName = null)
        {
            new FormatColumnConverter(formatPattern).Convert(dataTable, columnName, newColumnName);
        }

        /// <summary>
        /// 将 DataTable 转换为指定的模型集
        /// </summary>
        /// <param name="dataTable"></param>
        /// <typeparam name="TModel"></typeparam>
        /// <returns></returns>
        public static IEnumerable<TModel> ToModels<TModel>(this DataTable dataTable)
            where TModel : new()
        {
            if (dataTable == null)
            {
                return null;
            }

            return (from DataRow row in dataTable.Rows
                    select row.ToModel<TModel>());
        }
    }
}
